#coding:utf-8
import PIL.Image
import os

#图片压缩批处理  
def compressImage(srcPath,dstPath):  
    for filename in os.listdir(srcPath):  
        #如果不存在目的目录则创建一个，保持层级结构
        if not os.path.exists(dstPath):
                os.makedirs(dstPath)        

        #拼接完整的文件或文件夹路径
        srcFile=os.path.join(srcPath,filename)
        dstFile=os.path.join(dstPath,filename)
        print srcFile
        print dstFile

        #如果是文件就处理
        if os.path.isfile(srcFile):     
            try:
                #打开原图片缩小后保存，可以用if srcFile.endswith(".jpg")或者split，splitext等函数等针对特定文件压缩
                sImg=PIL.Image.open(srcFile)  
                w,h=sImg.size  
                print w,h
                dImg=sImg.resize((w/2,h/2),PIL.Image.ANTIALIAS)  #设置压缩尺寸和选项，注意尺寸要用括号
                dImg.save(dstFile) #也可以用srcFile原路径保存,或者更改后缀保存，save这个函数后面可以加压缩编码选项JPEG之类的
                print dstFile+" compressed succeeded"
            except IOError:
                print 'IOERROR' + srcFile

        #如果是文件夹就递归
        if os.path.isdir(srcFile):
            compressImage(srcFile,dstFile)

# 字节bytes转化kb\m\g
def formatSize(bytes):
    try:
        bytes = float(bytes)
        kb = bytes / 1024
    except:
        return "Error"

    if kb >= 1024:
        M = kb / 1024
        if M >= 1024:
            G = M / 1024
            return "%fG" % (G)
        else:
            return "%fM" % (M)
    else:
        return "%fkb" % (kb)

def infoImage(srcPath):
    for filename in os.listdir(srcPath):
        srcfile = os.path.join(srcPath,filename)
        if os.path.isfile(srcfile):
            try:
                sImg=PIL.Image.open(srcfile)
                # 获取照片宽高
                print sImg.size

                # 获取照片大小
                size = os.path.getsize(srcfile)
                print formatSize(size)
                
            except IOError:
                print srcPath

        if os.path.isdir(srcfile):
            infoImage(srcfile)

# 安装照片处理库: pip install pillow
if __name__=='__main__': 
    # infoImage("/Users/mac/Desktop/Files")
    compressImage("/Volumes/JunMing/Images/news","/Users/junming/Desktop/Document")












